#!/bin/bash

CODES_BUILD=~/codes_build
TORCH_VERSION=v2.5.0

mkdir -p ${CODES_BUILD}
cd ${CODES_BUILD}

# git clone git@github.com:pytorch/pytorch.git

cd pytorch
git checkout ${TORCH_VERSION}
git submodule sync
git submodule update --init --recursive

# -D CMAKE_CXX_STANDARD_REQUIRED=ON -D USE_CUDA=ON -D USE_CUDNN=ON \
# -D USE_OPENCV=OFF -D BUILD_CAFFE2_MOBILE=OFF -D BUILD_PYTHON=OFF -D BUILD_CAFFE2_OPS=OFF \
# -D BUILD_TEST=OFF -D USE_TBB=OFF \
mkdir -p build && cd build
cmake -D CMAKE_BUILD_TYPE=Debug -D CMAKE_INSTALL_PREFIX=/opt/libtorch -D CMAKE_CXX_STANDARD=17 \
	..

make -j16
